package com.example.utils;

import com.example.entity.ModelVersion;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;

/**
 * @Author: 张比特
 * @Data: 2024/12/2 上午10:04
 * @Version: v1.0.0
 * @Description: TODO
 **/
public class MongoDBUtil {
    @Autowired
    private MongoTemplate mongoTemplate;

    /**
     * 根据模型的版本查询模型
     * @param id
     * @return
     */
    public ModelVersion selectById(String id) {
        return mongoTemplate.findById(id, ModelVersion.class);
    }

    /**
     * 保存模型
     * @param modelVersion
     */
    public void save(ModelVersion modelVersion) {
        mongoTemplate.save(modelVersion);
    }
}